#ifndef TEXT_QUERY_H
#define TEXT_QUERY_H

#include <fstream>
#include <sstream>
#include <vector>
#include <string>
#include <memory>
#include <map>
#include <set>

class QueryResult;  // 为了定义函数 query 的返回类型，这个声明是必需的
class TextQuery {
public:
    using line_no = std::vector<std::string>::size_type;
    TextQuery(std::ifstream&);
    QueryResult query(const std::string&) const;
private:
    std::shared_ptr<std::vector<std::string>> file;  // 输入文件
    // 每个单词到它所在的行号的集合的映射
    std::map<std::string, std::shared_ptr<std::set<line_no>>> wm;
};

class QueryResult {
    friend std::ostream& print(std::ostream&, const QueryResult);
public:
    using line_no = std::vector<std::string>::size_type;
    QueryResult(std::string& s,
                std::shared_ptr<std::set<line_no>> p,
                std::shared_ptr<std::vector<std::string>> f)
        : sought(s), lines(p), file(f) { }
private:
    std::string &sought;  // 查询单词
    std::shared_ptr<std::set<line_no>> lines;  // 出现的行号
    std::shared_ptr<std::vector<std::string>> file;  // 输入文件
};

#endif // TEXT_QUERY_H
